﻿// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// Copyright (c) 2011, Dr. Masroor Ehsan
// All rights reserved.
// 
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above copyright
//       notice, this list of conditions and the following disclaimer in the
//       documentation and/or other materials provided with the distribution.
//     * Neither the name of the author nor the
//       names of its contributors may be used to endorse or promote products
//       derived from this software without specific prior written permission.
// 
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
// 
// $Author:$
// $Id: StockExchangeInfo.cs 21 2011-10-17 08:31:57Z masroore@gmail.com $
// $Rev:$
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *

using System;
using System.Collections.Generic;

namespace MaxTrader.IntradayQuotes.Core
{
    /// <summary>
    ///   Additional information regarding different stock exchanges (market open, close time)
    /// </summary>
    public static class StockExchangeInfo
    {
        private static readonly Dictionary<StockExchange, MarketHours> _marketHours;

        static StockExchangeInfo()
        {
            _marketHours = new Dictionary<StockExchange, MarketHours>();

            // TODO: load market timing from XML config
            //_marketHours.Add(StockExchange.Dse, new MarketHours(DateTime.UtcNow, DateTime.UtcNow));
            //_marketHours.Add(StockExchange.Cse, new MarketHours(DateTime.UtcNow, DateTime.UtcNow));
        }

        public static DateTime MarketOpenTimeUtc(StockExchange exchange)
        {
            return _marketHours[exchange].MarketOpenUtc;
        }

        public static DateTime MarketCloseTimeUtc(StockExchange exchange)
        {
            return _marketHours[exchange].MarketCloseUtc;
        }

        public static bool IsAfterMarketOpen(StockExchange exchange)
        {
            return IsAfterMarketOpen(exchange, DateTime.UtcNow);
        }

        public static bool IsAfterMarketOpen(StockExchange exchange, DateTime timeUtc)
        {
            var mktHr = _marketHours[exchange].MarketOpenUtc.Hour;
            var mktMin = _marketHours[exchange].MarketOpenUtc.Minute;

            var currHr = timeUtc.Hour;
            var currMin = timeUtc.Minute;

            return ((mktHr == currHr) && (currMin >= mktMin) || (currHr > mktHr));
        }

        public static bool IsBeforeMarketClose(StockExchange exchange)
        {
            return IsBeforeMarketClose(exchange, DateTime.UtcNow);
        }

        public static bool IsBeforeMarketClose(StockExchange exchange, DateTime timeUtc)
        {
            var mktHr = _marketHours[exchange].MarketOpenUtc.Hour;
            var mktMin = _marketHours[exchange].MarketOpenUtc.Minute;

            var currHr = timeUtc.Hour;
            var currMin = timeUtc.Minute;

            return ((mktHr == currHr) && (currMin <= mktMin) || (currHr < mktHr));
        }

        public static bool WithinTradingHours(StockExchange exchange)
        {
            return WithinTradingHours(exchange, DateTime.UtcNow);
        }

        public static bool WithinTradingHours(StockExchange exchange, DateTime timeUtc)
        {
            return (IsAfterMarketOpen(exchange, timeUtc) && IsBeforeMarketClose(exchange, timeUtc));
        }

        #region Nested type: MarketHours

        private struct MarketHours
        {
            public MarketHours(DateTime openUtc, DateTime closeUtc) : this()
            {
                MarketOpenUtc = openUtc;
                MarketCloseUtc = closeUtc;
            }

            internal DateTime MarketOpenUtc { get; private set; }
            internal DateTime MarketCloseUtc { get; private set; }
        }

        #endregion
    }
}